import numpy as np
def step_function(x):
    if x>0:
        return 1
    else :
        return 0
#这样写无法接受参数Numpy数组,if 数组>0会报错

def step_function(x):
    y=x>0     #[x1,x2,x3]>0->[true,flase,flase]
    return y.astype(int) #np.int已被弃用 astype用于指定转换的数字类型 
x=np.array([-1.0,1.0,2.0])
print(x)
y=x>0
print(y)
y=y.astype(int)  
print(y)